package cn.leecode.q35;

/**
 * 简单题目35:搜索插入位置
 * 给定一个排序数组和一个目标值，在数组中找到目标值，并返回其索引。如果目标值不存在于数组中，返回它将会被按顺序插入的位置。
 * 请必须使用时间复杂度为 O(log n) 的算法。
 * 示例 1:
 * 输入: nums = [1,3,5,6], target = 5
 * 输出: 2
 * 示例 2:
 * 输入: nums = [1,3,5,6], target = 2
 * 输出: 1
 * 示例 3:
 * 输入: nums = [1,3,5,6], target = 7
 * 输出: 4
 * 提示:
 * 1 <= nums.length <= 104
 * -104 <= nums[i] <= 104
 * nums 为 无重复元素 的 升序 排列数组
 * -104 <= target <= 104
 */
public class SearchInsertIndex {
    public int searchInsert(int[] nums, int target) {
        int left = 0;
        int right = nums.length - 1;
        while (left <= right) {
            int mid = left + (right - left) / 2; // 防止计算时溢出
            if (nums[mid] == target) {
                return mid; // 找到目标值，返回索引
            } else if (nums[mid] < target) {
                left = mid + 1; // 目标值大于中间值，搜索右半部分
            } else {
                right = mid - 1; // 目标值小于中间值，搜索左半部分
            }
        }
        // 如果没有找到目标值，`left` 就是要插入的位置。
        return left;
    }

    public static void main(String[] args) {
        SearchInsertIndex searchInsertIndex = new SearchInsertIndex();
        System.out.println(searchInsertIndex.searchInsert(new int[]{1,3,5,6}, 5)); // 输出: 2
        System.out.println(searchInsertIndex.searchInsert(new int[]{1,3,5,6}, 2)); // 输出: 1
        System.out.println(searchInsertIndex.searchInsert(new int[]{1,3,5,6}, 7)); // 输出: 4
    }
}
